package com.ruoyi.project.system.mapper;

import com.ruoyi.project.system.domain.SysUserRole;
import tk.mybatis.mapper.additional.insert.InsertListMapper;
import tk.mybatis.mapper.common.Mapper;
import tk.mybatis.mapper.entity.Example;

import java.util.Arrays;
import java.util.List;

/**
 * 用户与角色关联表 数据层
 * 
 * @author ruoyi
 */
public interface SysUserRoleMapper extends Mapper<SysUserRole>, InsertListMapper<SysUserRole>
{
    /**
     * 通过用户ID删除用户和角色关联
     * 
     * @param userId 用户ID
     * @return 结果
     */
    default int deleteUserRoleByUserId(Long userId) {
        Example example = new Example(SysUserRole.class);
        example.createCriteria().andEqualTo("userId", userId);
        return deleteByExample(example);
    }

    /**
     * 批量删除用户和角色关联
     * 
     * @param ids 需要删除的数据ID
     * @return 结果
     */
    default int deleteUserRole(Long[] ids) {
        Example example = new Example(SysUserRole.class);
        example.createCriteria().andIn("userId", Arrays.asList(ids));
        return deleteByExample(example);
    }

    /**
     * 通过角色ID查询角色使用数量
     * 
     * @param roleId 角色ID
     * @return 结果
     */
    default int countUserRoleByRoleId(Long roleId) {
        Example example = new Example(SysUserRole.class);
        example.createCriteria().andEqualTo("roleId", roleId);
        return selectCountByExample(example);
    }

    /**
     * 批量新增用户角色信息
     *
     * @param userRoleList 用户角色列表
     * @return 结果
     */
    default int batchUserRole(List<SysUserRole> userRoleList) {
        return insertList(userRoleList);
    }

    /**
     * 删除用户和角色关联信息
     * 
     * @param userRole 用户和角色关联信息
     * @return 结果
     */
    default int deleteUserRoleInfo(SysUserRole userRole) {
        Example example = new Example(SysUserRole.class);
        example.createCriteria()
                .andEqualTo("userId", userRole.getUserId())
                .andEqualTo("roleId", userRole.getRoleId());
        return deleteByExample(example);
    }

    /**
     * 批量取消授权用户角色
     * 
     * @param roleId 角色ID
     * @param userIds 需要删除的用户数据ID
     * @return 结果
     */
    default int deleteUserRoleInfos(Long roleId, Long[] userIds) {
        Example example = new Example(SysUserRole.class);
        example.createCriteria()
                .andEqualTo("roleId", roleId)
                .andIn("userId", Arrays.asList(userIds));
        return deleteByExample(example);
    }
}
